<script setup lang="ts">
// import type { VbenFormProps } from '@vben/common-ui';

import type { VxeGridProps } from '#/adapter/vxe-table';
import type { Region } from '#/api/system/region/model';

import { nextTick } from 'vue';

import { Page, useVbenDrawer } from '@vben/common-ui';
import { $t } from '@vben/locales';
import { eachTree, getVxePopupContainer } from '@vben/utils';

import { Popconfirm, Space } from 'ant-design-vue';

import { useVbenVxeGrid } from '#/adapter/vxe-table';
import { regionQueryByParentCode, regionRemove } from '#/api/system/region';

import { columns } from './data';
import regionDrawer from './region-drawer.vue';

// const formOptions: VbenFormProps = {
//   commonConfig: {
//     labelWidth: 80,
//     componentProps: {
//       allowClear: true,
//     },
//   },
//   schema: querySchema(),
//   wrapperClass: 'grid-cols-1 md:grid-cols-2 lg:grid-cols-3 xl:grid-cols-4',
// };

async function loadRegionByParentCode(code: string) {
  const resp = await regionQueryByParentCode(code);
  return resp.map((v) => {
    v.hasChild = true;
    return v;
  });
}

const gridOptions: VxeGridProps = {
  columns,
  height: 'auto',
  keepSource: true,
  pagerConfig: {
    enabled: false,
  },
  proxyConfig: {
    ajax: {
      query: async (_, values) => {
        const rows = await loadRegionByParentCode(values.parentCode);
        return { rows };
      },
      // // 默认请求接口后展开全部 不需要可以删除这段
      querySuccess: () => {
        // 默认全部不展开 需要加上标记
        eachTree(tableApi.grid.getData(), (item) => (item.expand = false));
        nextTick(() => {
          setExpandOrCollapse(false);
        });
      },
    },
  },
  /**
   * 虚拟滚动  默认关闭
   */
  scrollY: {
    enabled: false,
    gt: 0,
  },
  rowConfig: {
    keyField: 'code',
  },
  treeConfig: {
    parentField: 'parentCode',
    rowField: 'code',
    lazy: true,
    hasChildField: 'hasChild',
    // transform: true,
    loadMethod({ row }) {
      // 异步加载子节点
      return loadRegionByParentCode(row.code);
    },
  },
  id: 'system-region-index',
};

const [BasicTable, tableApi] = useVbenVxeGrid({
  // formOptions,
  gridOptions,
  gridEvents: {
    cellDblclick: (e) => {
      const { row = {} } = e;
      if (!row?.children) {
        return;
      }
      const isExpanded = row?.expand;
      tableApi.grid.setTreeExpand(row, !isExpanded);
      row.expand = !isExpanded;
    },
    // 需要监听使用箭头展开的情况 否则展开/折叠的数据不一致
    toggleTreeExpand: (e) => {
      const { row = {}, expanded } = e;
      row.expand = expanded;
    },
  },
});
const [RegionDrawer, drawerApi] = useVbenDrawer({
  connectedComponent: regionDrawer,
});

function handleAdd() {
  drawerApi.setData({
    update: false,
    sort: tableApi.grid?.getData()?.length + 1 || 1,
  });
  drawerApi.open();
}

function handleSubAdd(row: Region) {
  drawerApi.setData({ region: row, update: false });
  drawerApi.open();
}

async function handleEdit(row: Region) {
  drawerApi.setData({ region: row, update: true });
  drawerApi.open();
}

async function handleDelete(row: Region) {
  await regionRemove([row.code]);
  await tableApi.query();
}

/**
 * 全部展开/折叠
 * @param expand 是否展开
 */
function setExpandOrCollapse(expand: boolean) {
  eachTree(tableApi.grid.getData(), (item) => (item.expand = expand));
  if (expand) {
    tableApi.grid?.setAllTreeExpand(expand);
  } else {
    tableApi.grid?.clearTreeExpand();
  }
}
</script>

<template>
  <Page :auto-content-height="true">
    <BasicTable>
      <template #table-title>
        <main-title text="区划列表" help="双击展开/收起子菜单" />
      </template>
      <template #toolbar-tools>
        <Space>
          <!--<a-button @click="setExpandOrCollapse(false)">-->
          <!--  {{ $t('pages.common.collapse') }}-->
          <!--</a-button>-->
          <!--<a-button @click="setExpandOrCollapse(true)">-->
          <!--  {{ $t('pages.common.expand') }}-->
          <!--</a-button>-->
          <a-button
            type="primary"
            v-access:code="['system:region:add']"
            @click="handleAdd"
          >
            {{ $t('pages.common.add') }}
          </a-button>
        </Space>
      </template>
      <template #action="{ row }">
        <Space>
          <ghost-button
            v-access:code="['system:region:edit']"
            @click="handleEdit(row)"
          >
            {{ $t('pages.common.edit') }}
          </ghost-button>
          <ghost-button
            class="btn-success"
            v-access:code="['system:region:add']"
            @click="handleSubAdd(row)"
          >
            {{ $t('pages.common.add') }}
          </ghost-button>
          <Popconfirm
            :get-popup-container="getVxePopupContainer"
            placement="left"
            title="确认删除？"
            @confirm="handleDelete(row)"
          >
            <ghost-button
              danger
              v-access:code="['system:region:remove']"
              @click.stop=""
            >
              {{ $t('pages.common.delete') }}
            </ghost-button>
          </Popconfirm>
        </Space>
      </template>
    </BasicTable>
    <RegionDrawer @reload="tableApi.query()" />
  </Page>
</template>
